import torch
import numpy as np
from joblib import Parallel, delayed

def create_all_data_pytorch(batch_size, data_len, ini_pos_noise, ini_vel_noise, batch_num, device=None):
    """
    使用PyTorch加速生成轨迹数据
    参数与原函数相同，增加了
    :param device ： PyTorch设备('cuda','cpu'或None为自动检测)
    """
    if device is None:
        device = torch.device('cuda' if torch.)